package com.jxb.tree;
import java.util.ArrayList;
import java.util.Deque;
import java.util.LinkedList;
import java.util.List;

/**
 * 中序排序：从根节点开始，寻找左子树，一直找下去
 *  若左子树为null，将当前节点的值设置到集合中，然后再获取右子树，若右子树为null，将值设置到集合中
 *  然后再依次执行
 */
public class BinTreeInOrderTraversal_94 {

    public static void main(String[] args) {

    }

    //递归-时间复杂度为O(n)
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        accessTree(root,res);
        return res;
    }

    public void accessTree(TreeNode root, List<Integer> res) {
        if (root == null) {
            return;
        }
        accessTree(root.left,res);
        res.add(root.val);
        accessTree(root.right,res);
    }

    //栈，实现逻辑如下：
    //1：将所有左子树放入栈中
    //2：出栈，获取最后一个子树，获取val值
    //3：获取子树的右子树，将右子树压入栈中
    //4：将右子树的所有左子树压入栈中，然后出栈，获取val值
    public List<Integer> inorderTraversalWithLoop(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        Deque<TreeNode> stack = new LinkedList<>();
        while (root != null || !stack.isEmpty()) {
            while (root != null) {
                stack.push(root);
                root = root.left;
            }
            root = stack.pop();
            res.add(root.val);
            root = root.right;
        }
        return res;
    }


}
